package com.pkche.core.advanced.qrcode;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
 



 import javax.imageio.ImageIO;

import org.apache.log4j.Logger;

import java.io.File;
import java.io.OutputStream;
import java.io.IOException;
import java.util.Hashtable;
import java.awt.image.BufferedImage;
 
 
 public final class MatrixToImageWriter {
	
	private static Logger logger = Logger.getLogger(MatrixToImageWriter.class);
 
   private static final int BLACK = 0xFF000000;
   private static final int WHITE = 0xFFFFFFFF;
 
   private MatrixToImageWriter() {}
 
   
   public static BufferedImage toBufferedImage(BitMatrix matrix) {
     int width = matrix.getWidth();
     int height = matrix.getHeight();
     BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
     for (int x = 0; x < width; x++) {
       for (int y = 0; y < height; y++) {
         image.setRGB(x, y, matrix.get(x, y) ? BLACK : WHITE);
       }
     }
     return image;
   }
 
   
   public static void writeToFile(BitMatrix matrix, String format, File file)
       throws IOException {
     BufferedImage image = toBufferedImage(matrix);
     if (!ImageIO.write(image, format, file)) {
       throw new IOException("Could not write an image of format " + format + " to " + file);
     }
   }
 
   
   public static void writeToStream(BitMatrix matrix, String format, OutputStream stream)
       throws IOException {
	     BufferedImage image = toBufferedImage(matrix);
	     if (!ImageIO.write(image, format, stream)) {
	       throw new IOException("Could not write an image of format " + format);
	     }
   }
   
   /**
    * 生成URL链接二维码
    * @param text
    * @param path
    */
	   public static void createURLImage(String text,String path){
		   long start = System.currentTimeMillis();
		   logger.debug("开始生成二维码图片 : text:" +text +"   \n path:"+path);
	        int width = 300;  //360*200为微信支持的图片
	        int height = 180;  
	        //二维码的图片格式  
	        String format = "gif";  
	        Hashtable  hints = new Hashtable();  
	        //内容所使用编码  
	        hints.put(EncodeHintType.CHARACTER_SET, "utf-8");  
	        BitMatrix bitMatrix;
			try {
				bitMatrix = new MultiFormatWriter().encode(text,  
				        BarcodeFormat.QR_CODE, width, height, hints);
				   File outputFile = new File(path);  
			        try {
						writeToFile(bitMatrix, format, outputFile);
					} catch (IOException e) {
						e.printStackTrace();
					}  
			} catch (WriterException e) {
				logger.error("生成二维码图片发生错误",e);
				e.printStackTrace();
			}  
		   logger.debug("生成二维码图片结束 ， 共耗时："
			+(System.currentTimeMillis() - start) +" 毫秒");
	   }
 
 }